package roleController

import (
	"myRbac/models"
	"myRbac/utils"
	"net/http"
	"strconv"
	"time"

	"github.com/gin-gonic/gin"
)

// 定义角色结构体
type RoleController struct{}

// @Summary 角色查询
// @Description 分页查询角色
// @Tags 角色管理
// @Accept json
// @Produce json
// @Param page query string false "页码"
// @Param pageSize query string false "尺寸"
// @Param name query string false "角色名"
// @Success 200 {object} models.ResultByPage[models.Role] "查询成功"
// @Router /role [get]
// @Security JWT
func (r RoleController) QueryRoles(ctx *gin.Context) {
	// 获取参数
	page, _ := strconv.Atoi(ctx.DefaultQuery("page", "1"))
	pageSize, _ := strconv.Atoi(ctx.DefaultQuery("pageSize", "10"))
	nameStr := ctx.DefaultQuery("name", "")
	name := "%" + nameStr + "%"
	// 承载角色
	roleList := []models.Role{}
	models.DB.Where("is_delete!=1 AND name like (?)", name).Limit(pageSize).Offset(pageSize * (page - 1)).Order("id").Find(&roleList)
	var total int64
	// 查询条目
	models.DB.Table("roles").Where("is_delete!=1 AND name like (?)", name).Count(&total)
	// 返回值约束
	result := &models.ResultByPage[models.Role]{
		Msg:      "ok",
		Code:     "200",
		Total:    total,
		Page:     page,
		PageSize: pageSize,
		Result:   roleList,
	}
	ctx.JSON(http.StatusOK, result)
}

// @Summary 角色查询
// @Description 查询所有角色
// @Tags 角色管理
// @Accept json
// @Produce json
// @Success 200 {object} models.ResultCommon[[]models.Role] "查询成功"
// @Router /role/all [get]
// @Security JWT
func (r RoleController) QueryAllRoles(ctx *gin.Context) {
	// 承载角色
	roleList := []models.Role{}
	models.DB.Where("is_delete!=1").Find(&roleList)
	// 返回值约束
	result := &models.ResultCommon[[]models.Role]{
		Msg:    "ok",
		Code:   "200",
		Result: roleList,
	}
	ctx.JSON(http.StatusOK, result)
}

// @Summary 角色查询
// @Description 根据Id查询角色
// @Tags 角色管理
// @Accept json
// @Produce json
// @Param id path string true "角色id"
// @Success 200 {object} models.ResultCommon[models.Role] "查询成功"
// @Success 401 {object} models.ResultCommon[string] "查询失败"
// @Router /role/{id} [get]
// @Security JWT
func (r RoleController) QueryRoleById(ctx *gin.Context) {
	role := &models.Role{}
	role_id := ctx.Param("id")
	models.DB.Where("id=?", role_id).First(&role)
	if role.Id == 0 {
		ctx.JSON(http.StatusOK, &models.ResultCommon[string]{
			Msg:    "查询失败",
			Code:   "400",
			Result: "",
		})
	} else {
		ctx.JSON(http.StatusOK, &models.ResultCommon[models.Role]{
			Msg:    "查询成功",
			Code:   "200",
			Result: *role,
		})
	}
}

// @Summary 新增角色
// @Description 传入json，新增角色
// @Tags 角色管理
// @Accept json
// @Produce json
// @Param roleInfo body models.Role true "角色信息"
// @Success 200 {object} models.AddResult "新增成功"
// @fail 400 {object} models.ErrorResult "数据错误"
// @fail 401 {object} models.AddResult "新增失败"
// @Router /role [post]
// @Security JWT
func (r RoleController) AddRole(ctx *gin.Context) {
	role := &models.Role{}
	// 绑定json数据
	if err := ctx.ShouldBindJSON(&role); err != nil {
		ctx.JSON(http.StatusOK, &models.ErrorResult{
			Msg:    "参数异常",
			Code:   "400",
			Result: err,
		})
	} else {
		role.CreateTime = utils.TimeFormatter(time.Now())
		result := models.DB.Model(&models.Role{}).Create(&role)
		if result.RowsAffected != 1 {
			ctx.JSON(http.StatusOK, &models.AddResult{
				Msg:    "新增失败",
				Code:   "400",
				Result: result.RowsAffected,
			})
		} else {
			ctx.JSON(http.StatusOK, &models.AddResult{
				Msg:    "新增成功",
				Code:   "200",
				Result: result.RowsAffected,
			})
		}
	}
}

// @Summary 更新角色
// @Description 传入json，更新角色
// @Tags 角色管理
// @Accept json
// @Produce json
// @Param roleInfo body models.Role true "角色信息"
// @Success 200 {object} models.UpdateResult[models.Role] "更新成功"
// @fail 400 {object} models.ErrorResult "数据错误"
// @fail 401 {object} models.ErrorResult "角色不存在"
// @fail 402 {object} models.AddResult "更新失败"
// @Router /role [put]
// @Security JWT
func (r RoleController) UpdateRole(ctx *gin.Context) {
	// 获取提交的角色修改信息
	var role models.Role
	// 绑定json数据
	if err := ctx.ShouldBindJSON(&role); err != nil {
		ctx.JSON(http.StatusOK, &models.ErrorResult{
			Msg:    "数据绑定失败",
			Code:   "400",
			Result: err,
		})
		return
	}
	// 绑定角色信息
	role_old := &models.Role{}
	// 查找角色信息
	models.DB.First(&role_old, role.Id)
	if role_old.Id == 0 {
		ctx.JSON(http.StatusOK, &models.ErrorResult{
			Msg:    "角色不存在",
			Code:   "400",
			Result: nil,
		})
	}
	result := models.DB.Model(&role_old).Updates(role)
	if result.RowsAffected != 1 {
		ctx.JSON(http.StatusOK, &models.ErrorResult{
			Msg:    "更新失败",
			Code:   "400",
			Result: result.Error,
		})
	} else {
		ctx.JSON(http.StatusOK, &models.UpdateResult[models.Role]{
			Msg:    "更新成功",
			Code:   "200",
			Result: role,
		})
	}
}

// @Summary 删除角色
// @Description 传入json，批量删除角色
// @Tags 角色管理
// @Accept json
// @Produce json
// @Param roleIds body models.DeleteParams true "角色id合集"
// @Success 200 {object} models.DeleteResult "更新成功"
// @fail 400 {object} models.ErrorResult "数据错误"
// @fail 401 {object} models.DeleteResult "删除失败"
// @Router /role [delete]
// @Security JWT
func (r RoleController) DeleteRole(ctx *gin.Context) {
	// 绑定参数实例
	ids := &models.DeleteParams{}
	// 定义模型实例
	var role models.Role
	if err := ctx.ShouldBindJSON(&ids); err != nil {
		ctx.JSON(http.StatusOK, &models.ErrorResult{
			Msg:    "数据绑定失败",
			Code:   "400",
			Result: err,
		})
	} else {
		// 真的删
		// result := models.DB.Where("id IN (?)", ids.Ids).Delete(&role)
		// 假的删
		result := models.DB.Model(&role).Where("id IN (?)", ids.Ids).Update("is_delete", 1)
		if result.RowsAffected != 0 {
			ctx.JSON(200, &models.DeleteResult{
				Msg:    "删除成功",
				Code:   "200",
				Result: ids.Ids,
			})
		} else {
			ctx.JSON(200, &models.DeleteResult{
				Msg:    "删除失败",
				Code:   "400",
				Result: []int{0},
			})
		}
	}
}
